#!/bin/bash

# 设置日志级别: DEBUG(详细调试信息), INFO(基本信息), WARNING(警告), ERROR(仅错误)
export RFDIFFUSION_LOG_LEVEL=INFO

# 强制使用第一块NPU (npu:0)
export ASCEND_RT_VISIBLE_DEVICES=0
export CUDA_VISIBLE_DEVICES=0

# 设置图后端为PyG (根据需要可修改)
export GRAPH_BACKEND=pyg

# 运行推理脚本，指定参数
# ./scripts/run_inference.py \
#     'contigmap.contigs=[200-200]' \
#     inference.output_prefix=test_outputs_200_50/test \
#     inference.num_designs=50

./scripts/run_inference.py \
    'contigmap.contigs=[400-400]' \
    inference.output_prefix=a/test \
    inference.num_designs=50

./scripts/run_inference.py \
    'contigmap.contigs=[600-600]' \
    inference.output_prefix=test_outputs_600_50/test \
    inference.num_designs=50

./scripts/run_inference.py \
    'contigmap.contigs=[800-800]' \
    inference.output_prefix=test_outputs_800_50/test \
    inference.num_designs=50

./scripts/run_inference.py \
    'contigmap.contigs=[1000-1000]' \
    inference.output_prefix=test_outputs_1000_50/test \
    inference.num_designs=50
# python scripts/run_inference.py \
#         'contigmap.contigs=[50-50]' \
#         inference.num_designs=1 \
#         inference.deterministic=True \
#         diffuser.T=15 \
#         inference.final_step=8 \
#         inference.output_prefix=test_pyg_outputs_3/test

# rm -rf test_outputs_